package com.token.iot.bratterycharging.boss.qrcode;

import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.util.HashMap;
import java.util.Hashtable;
import java.util.Map;

import javax.imageio.ImageIO;

import org.apache.commons.codec.binary.Base64;
import org.apache.commons.codec.binary.Base64OutputStream;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
import com.outstanding.framework.utils.Base64Util;

public class QrCodeUtils {

	public static final String QRCODE_DEFAULT_CHARSET = "UTF-8";
	
	//图片的高度
	public static final int QRCODE_DEFAULT_HEIGHT = 280;
	//图片的宽度

	public static final int QRCODE_DEFAULT_WIDTH = 260;

	private static final int BLACK = 0xFF000000;
	private static final int WHITE = 0xFFFFFFFF;

	/**
	 * Create qrcode with default settings
	 *
	 * @param data
	 * @return
	 */
	public static BufferedImage createQRCode(String data) {
		return createQRCode(data, QRCODE_DEFAULT_WIDTH, QRCODE_DEFAULT_HEIGHT);
	}

	/**
	 * Create qrcode with default charset
	 *
	 * 
	 * @param data
	 * @param width
	 * @param height
	 * @return
	 */
	public static BufferedImage createQRCode(String data, int width, int height) {
		return createQRCode(data, QRCODE_DEFAULT_CHARSET, width, height);
	}

	/**
	 * Create qrcode with specified charset
	 *
	 * @param data
	 * @param charset
	 * @param width
	 * @param height
	 * @return
	 */
	@SuppressWarnings({ "unchecked", "rawtypes" })
	public static BufferedImage createQRCode(String data, String charset, int width, int height) {
		Map hint = new HashMap();
		hint.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
		hint.put(EncodeHintType.CHARACTER_SET, charset);

		return createQRCode(data, charset, hint, width, height);
	}

	/**
	 * Create qrcode with specified hint
	 *
	 * @param data
	 * @param charset
	 * @param hint
	 * @param width
	 * @param height
	 * @return
	 */
	public static BufferedImage createQRCode(String data, String charset, Map<EncodeHintType, ?> hint, int width,
			int height) {
		BitMatrix matrix;
		try {
			matrix = new MultiFormatWriter().encode(new String(data.getBytes(charset), charset), BarcodeFormat.QR_CODE,
					width, height, hint);
			return toBufferedImage(matrix);
		} catch (WriterException e) {
			throw new RuntimeException(e.getMessage(), e);
		} catch (Exception e) {
			throw new RuntimeException(e.getMessage(), e);
		}
	}

	public static BufferedImage toBufferedImage(BitMatrix matrix) {
		int width = matrix.getWidth();
		int height = matrix.getHeight();
		BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
		for (int x = 0; x < width; x++) {
			for (int y = 0; y < height; y++) {
				image.setRGB(x, y, matrix.get(x, y) ? BLACK : WHITE);
			}
		}
		return image;
	}

	/**
	 * Create qrcode with default settings and logo
	 *
	 * 
	 * @param data
	 * @param logoFile
	 * @return
	 */
	public static BufferedImage createQRCodeWithLogo(String data, File logoFile) {
		return createQRCodeWithLogo(data, QRCODE_DEFAULT_WIDTH, QRCODE_DEFAULT_HEIGHT, logoFile);
	}

	/**
	 * Create qrcode with default charset and logo
	 *
	 * 
	 * @param data
	 * @param width
	 * @param height
	 * @param logoFile
	 * @return
	 */
	public static BufferedImage createQRCodeWithLogo(String data, int width, int height, File logoFile) {
		return createQRCodeWithLogo(data, QRCODE_DEFAULT_CHARSET, width, height, logoFile);
	}

	/**
	 * Create qrcode with specified charset and logo
	 *
	 * 
	 * @param data
	 * @param charset
	 * @param width
	 * @param height
	 * @param logoFile
	 * @return
	 */
	@SuppressWarnings({ "unchecked", "rawtypes" })
	public static BufferedImage createQRCodeWithLogo(String data, String charset, int width, int height,
			File logoFile) {
		Map hint = new HashMap();
		hint.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
		hint.put(EncodeHintType.CHARACTER_SET, charset);

		return createQRCodeWithLogo(data, charset, hint, width, height, logoFile);
	}

	/**
	 * Create qrcode with specified hint and logo
	 *
	 * 
	 * @param data
	 * @param charset
	 * @param hint
	 * @param width
	 * @param height
	 * @param logoFile
	 * @return
	 */
	public static BufferedImage createQRCodeWithLogo(String data, String charset, Map<EncodeHintType, ?> hint,
			int width, int height, File logoFile) {
		try {
			BufferedImage qrcode = createQRCode(data, charset, hint, width, height);
			BufferedImage logo = ImageIO.read(logoFile);
			int deltaHeight = height - logo.getHeight();
			int deltaWidth = width - logo.getWidth();

			BufferedImage combined = new BufferedImage(height, width, BufferedImage.TYPE_INT_ARGB);
			Graphics2D g = (Graphics2D) combined.getGraphics();
			g.drawImage(qrcode, 0, 0, null);
			g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER, 1f));
			g.drawImage(logo, (int) Math.round(deltaWidth / 2), (int) Math.round(deltaHeight / 2), null);

			return combined;
		} catch (IOException e) {
			throw new RuntimeException(e.getMessage(), e);
		} catch (Exception e) {
			throw new RuntimeException(e.getMessage(), e);
		}
	}

	/**
	 * Return base64 for image
	 *
	 * 
	 * @param image
	 * @return
	 */
	public static String getImageBase64String(BufferedImage image) {
		String result = null;
		try {
			ByteArrayOutputStream os = new ByteArrayOutputStream();
			OutputStream b64 = new Base64OutputStream(os);
			ImageIO.write(image, "png", b64);
			result = os.toString("UTF-8");
		} catch (UnsupportedEncodingException e) {
			throw new RuntimeException(e.getMessage(), e);
		} catch (IOException e) {
			throw new RuntimeException(e.getMessage(), e);
		}
		return result;
	}

	/**
	 * Decode the base64Image data to image
	 *
	 * @param base64ImageString
	 * @param file
	 */
	public static void convertBase64StringToImage(String base64ImageString, File file) {
		FileOutputStream os;
		try {
			Base64 d = new Base64();
			byte[] bs = d.decode(base64ImageString);
			os = new FileOutputStream(file.getAbsolutePath());
			os.write(bs);
			os.close();
		} catch (FileNotFoundException e) {
			throw new RuntimeException(e.getMessage(), e);
		} catch (IOException e) {
			throw new RuntimeException(e.getMessage(), e);
		} catch (Exception e) {
			throw new RuntimeException(e.getMessage(), e);
		}
	}
	
	
	/**
	 * @param pressText
	 *            文字
	 * @param image
	 *            需要添加文字的图片
	 * @param fontStyle
	 * @param color
	 * @param fontSize
	 * @param width
	 * @param height
	 * @为图片添加文字
	 */
	public static void pressText(String pressText,BufferedImage image) {
		int font = 16; //字体大小
        pressText(pressText,image,Color.black,font,QRCODE_DEFAULT_WIDTH,QRCODE_DEFAULT_HEIGHT);
	}
	
	/**
	 * @param pressText
	 *            文字
	 * @param image
	 *            需要添加文字的图片
	 * @param color
	 * @param fontSize
	 * @param width
	 * @param height
	 * @为图片添加文字
	 */
	public static void pressText(String pressText,BufferedImage image, Color color,
			int fontSize, int width, int height) {
		// 计算文字开始的位置
		// x开始的位置：（图片宽度-字体大小*字的个数）/2
		int startX = 40;
		//System.out.println(startX);
		// y开始的位置：图片高度-（图片高度-图片宽度）/2
		int startY = height - (height - width) / 2 -13 ;
		System.out.println(startY);
		try {
			int imageW = image.getWidth();
			int imageH = image.getHeight();
			
			Graphics g = image.createGraphics();
			g.drawImage(image, 0, 0, imageW, imageH, null);
			g.setColor(color);
			g.setFont(new Font("宋体", Font.PLAIN, fontSize));
			g.drawString(pressText, startX, startY);
			g.dispose();

			
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public static void main(String[] args) {
		File logfile  =new File("/Users/chenzhixuan/Documents/充电未来/logo1.png");
		BufferedImage image = QrCodeUtils.createQRCodeWithLogo("http://wechat.ifml.me/#/device?imei=866289033735384",logfile);
		QrCodeUtils.pressText("866289033735384", image);
		// try {
		// ImageIO.write(image, "png", new
		// File("/Users/chenzhixuan/Desktop/result7.png"));
		// } catch (IOException e) {
		// TODO Auto-generated catch block
		// e.printStackTrace();
		// }
		String str = getImageBase64String(image);
		System.out.println(str);
		File f = new File("/Users/chenzhixuan/Desktop/result9.png");
		convertBase64StringToImage(str, f);
	}
}
